import numpy as np
import torch
import random
import math
import torch.nn.functional as F
import matplotlib.pyplot as plt

a = torch.rand(1, 1280, 1280)
# https://blog.csdn.net/jjw_zyfx/article/details/120392966
print(a.squeeze().shape)  # 即将一维的去掉
